#pragma once
#include <iostream>
#include <string>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/time.h>
#include <unistd.h>
#include <atomic>
#include <fstream>
#include <vector>
#include <boost/algorithm/string.hpp>

namespace ns_util
{
    const std::string temp_path = "./temp/"; // 公共编译路径
    class TimeUtil
    {
    public:
        static std::string GetTimeStamp()
        {
            struct timeval tv;
            gettimeofday(&tv, nullptr); // 获取时间,时区(忽略)
            return std::to_string(tv.tv_sec);
        }
        static std::string GetTimeMS() // 获取毫秒
        {
            struct timeval tv;
            gettimeofday(&tv, nullptr);
            return std::to_string(tv.tv_sec * 1000 + tv.tv_usec / 1000);
        }
    };
    class PathUtil
    {
        // 对文件路径处理工具，为编译模块提供文件名处理
    public:
        static std::string AddSuffix(const std::string &file_name, const std::string &suffix)
        {
            std::string path_name = temp_path;
            path_name += file_name;
            path_name += suffix;
            return path_name;
        }
        // 编译相关文件
        static std::string Src(const std::string &file_name)
        {
            return AddSuffix(file_name, ".cpp");
        }
        static std::string Exec(const std::string &file_name)
        {
            return AddSuffix(file_name, ".exe");
        }
        static std::string Compile_error(const std::string &file_name)
        {
            return AddSuffix(file_name, ".compiler_error");
        }
        // 运行相关文件(输入，输出，错误)
        static std::string Stdin(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdin");
        }
        static std::string Stdout(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdout");
        }
        static std::string Stderr(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stderr");
        }
    };
    class FileUtil
    {
    public:
        static bool IsFileExists(const std::string &path_name)
        {
            struct stat st;
            // stat函数获取文件信息，获取成功返回0
            if (stat(path_name.c_str(), &st) == 0)
            {
                return true;
            }
            return false;
        }
        static std::string UniqueFileName()
        {
            static std::atomic_uint id(0);
            id++;
            std::string ms = TimeUtil::GetTimeMS();
            std::string uniq_id = std::to_string(id);
            return ms + "_" + uniq_id;
        }
        static bool WriteFile(const std::string &file_name, const std::string &content)
        {
            std::ofstream out(file_name);
            if (!out.is_open())
            {
                return false;
            }
            out.write(content.c_str(), content.size());
            out.close();
            return true;
        }
        static bool ReadFile(const std::string &target, std::string *content, bool keep = false)
        {
            (*content).clear();
            std::ifstream in(target);
            if (!in.is_open())
            {
                return false;
            }
            std::string line;
            // 按行读取，getline会忽略掉\n
            while (getline(in, line))
            {
                (*content) += line;
                (*content) += (keep ? "\n" : "");
            }
            in.close();
            return true;
        }
    };
    class StringUtil
    {
    public:
        static void SplitString(const std::string &src, std::vector<std::string> *target, const std::string &sep)
        {
            // 传入参数：结果目标，源字符串，分割符（is_any_of遇到分割符就切割），是否压缩（on表示压缩；off表示不压缩，不合并）
            boost::split((*target), src, boost::is_any_of(sep), boost::algorithm::token_compress_on);
        }
    };
}